../../../../../models/experimental/stable_diffusion_xl_base/vae/tests/pcc/test_module_tt_midblock2d.py